from langchain.prompts import PromptTemplate


# 小说背景设定
setting_prompt = PromptTemplate(
    input_variables=["genre", "theme"],
    template="请为一部{genre}小说生成世界观设定，核心主题是{theme}。要求包含时代背景、关键地点和基础矛盾。"
)

# 角色生成模板
character_prompt = PromptTemplate(
    input_variables=["setting", "role_type"],
    template="基于以下世界观：{setting}\n\n请设计一个{role_type}角色，需包含姓名、性格特征、秘密目标和经典台词。"
)

# 情节推进模板
plot_prompt = PromptTemplate(
    input_variables=["setting", "characters", "current_plot"],
    template="""
    世界观：{setting}
    已存在角色：{characters}
    当前剧情：{current_plot}

    请生成三个后续情节选项，每个需包含：
    1. 冲突类型（如背叛、意外发现等）
    2. 关键转折点
    3. 对至少两个角色的影响
    格式：选项1：[冲突]-[转折]-[影响]；选项2...；选项3...
    """
)